Java8 stream 中利用 groupingBy 进行多字段分组

2024-07-17 11:47| 来源: 网络整理| 查看: 265

从简单入手

Stream 作为 Java 8 的一大亮点，好比一个高级的迭代器（Iterator），单向，不可往复，数据只能遍历一次，遍历过一次后即用尽了，就好比流水从面前流过，一去不复返。

Java 8 中的 Streams API 详解

我们可以利用stream对数据进行分组。示例如下：

List items = Arrays.asList("apple", "apple", "banana", "apple", "orange", "banana", "papaya"); Map result = items.stream().collect( Collectors.groupingBy( Function.identity(), Collectors.counting() ) ); System.out.println(result);

输出如下：

{papaya=1, orange=1, banana=2, apple=3} 进阶需求

在实际需求中，我们可能需要对一组对象进行分组，而且分组的条件有多个。比如对国家和产品类型进行双重分组，一种比较复杂的方法是先对产品类型进行分组，然后对每一个产品类型中的国家名进行分组求和。示例如下：

Map countMap = recordItems.stream().collect(Collectors.groupingBy(o -> o.getProductType())); List records = new ArrayList; countMap.keySet().forEach(productType -> { Map countMap1 = countMap.get(productType).stream().collect(Collectors.groupingBy(o -> o.getCountry(), Collectors.counting())); countMap1(key).stream().forEach(country -> { Record record = new Record(); record.set("device_type", productType); record.set("location", country; record.set("count", countMap1(key).intValue()); records.add(record); }); }); 更好的解决方法

上面的方法在应对两个字段的分组要求时，还能应付的过来，但如果字段超过两个时，每增加一个字段，就会多出很多代码行，显然不太合理。更合理的方法是，增加一个 getKey()方法，返回多个字段组成的唯一key，比如通过下划线连接等等。示例如下：

// 分组统计 Map countMap = records.stream().collect(Collectors.groupingBy(o -> o.getProductType() + "_" + o.getCountry(), Collectors.counting())); List countRecords = countMap.keySet().stream().map(key -> { String[] temp = key.split("_"); String productType = temp[0]; String country = temp[1]; Record record = new Record(); record.set("device_type", productType); record.set("location", country; record.set("count", countMap.get(key).intValue()); return record; }).collect(Collectors.toList()); 参考资料 Java 8 中的 Streams API 详解 Java 8 – 分组GroupBy

文章标题：Java8 stream 中利用 groupingBy 进行多字段分组文章作者：Ciel Ni 文章链接：http://www.cielni.com/2018/07/14/java-stream-groupingby/ 有问题或建议欢迎与我联系讨论，转载或引用希望标明出处，感激不尽！

【本文地址】

公司简介

联系我们

今日新闻

推荐新闻

专题文章